# 数据集文件列表
DATA_FILES = [
    'final_track2_train.txt',
    'final_track2_test_no_anwser.txt',
    'track2_face_attrs.txt',
    'track2_audio_features.txt',
    'track2_title.txt',
    'track2_video_features.txt'
]
# 定义训练集和测试集的表头
TRAIN_TEST_COLUMNS = [
        'uid', 'user_city', 'item_id', 'author_id', 'item_city',
        'channel', 'finish', 'like', 'music_id', 'did',
        'creat_time', 'video_duration'
]